package bintree.leetcode;

//后序遍历的迭代写法

import java.util.ArrayList;
import java.util.Deque;
import java.util.LinkedList;
import java.util.List;

public class Num145_PostOrderNonRecursion {
    public List<Integer> postorderTraversal(TreeNode root) {
        List<Integer> ret=new ArrayList<>();
        if(root==null){
            return ret;
        }

        Deque<TreeNode> stack=new LinkedList<>();
        TreeNode cur=root;
        // 上一个完全处理过的节点(左右根都处理完毕的节点)
        TreeNode prev=null;
        while (!stack.isEmpty() || cur!=null){
            while (cur!=null){
                stack.push(cur);
                cur=cur.left;
            }

            // 此时左树为空，cur取栈顶元素，第二次访问
            cur=stack.pop();
            // 判断右树是否为空或者被我们访问过
            if(cur.right==null || prev==cur.right){
                ret.add(cur.val);
                // 当前节点cur就是最后处理的根节点，更新prev引用，变为cur
                prev=cur;
                cur=null;
            }else {
                // 此时右树不为空且没有处理过,就需要把根节点再压入栈中，继续处理右子树
                stack.push(cur);
                cur=cur.right;
            }
        }
        return ret;
    }
}
